package com.powernode.mybatis.utils;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

import java.io.IOException;

public class SqlSessionUtils {
    //私有化构造方法，防止new对象，因为工具类可以直接类名.调用
    private SqlSessionUtils(){}

    private static SqlSessionFactory sqlSessionFactory;

    // 全局的，服务器级别的，一个服务器当中定义一个即可
    // 为什么把SqlSession对象放到ThreadLocal当中呢？为了保证一个线程对应一个SqlSession对象
    private static ThreadLocal<SqlSession> local = new ThreadLocal<>();

    /**
     * 类加载时初始化SqlSessionFactory对象
     */
    static {
        try {
            sqlSessionFactory = new SqlSessionFactoryBuilder().build(Resources.getResourceAsStream("mybatis-config.xml"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取会话对象
     * @return 会话对象
     */
    public static SqlSession openSession(){
        //从当前线程中获取sqlSession对象
        SqlSession sqlSession = local.get();
        //第一次获取为空就new一个
        if (sqlSession == null) {
            sqlSession = sqlSessionFactory.openSession();
            //将SqlSession对象绑定到当前线程中
            local.set(sqlSession);
        }
        //返回
        return sqlSession;
    }

    /**
     * 关闭SqlSession对象(从当前线程中移除SqlSession对象)
     * @param sqlSession
     */
    public static void close(SqlSession sqlSession){
        if (sqlSession != null) {
            sqlSession.close();
            //注意移除SqlSession对象和当前线程的绑定关系
            //因为Tomcat服务器支持多线程，也就是说，用过的线程t1，可能下一次还会使用这个t1线程
            local.remove();
        }
    }
}
